# **********************OBJ INFO**************************
# Author:YuKalix
# @Time    : 2019-9-21 1:59
# @Site    : 52ziyu.cn
# @File    : FileManage.py
# @software: PyCharm
# *********************************************************
import os


class FileManage:

    def get_file_path_list_from_dir(self, dir_path):
        file_path_list = []
        file_name = []
        for dir, j, file_list in os.walk(dir_path):
            if file_list:
                for file in file_list:
                    file_path = os.path.join(dir, file)
                    file_path_list.append(file_path)
                    file_name.append(file)
        return file_path_list, file_name

    def write_json(self, filename):
        with open(filename) as f:
            f.write();


if __name__ == '__main__':
    S = FileManage()
    file_path_list = S.get_file_path_list_from_dir(r'G:\graduation_project\data_set\training_set')
    print(file_path_list)
